azalia mirhoseini
That Chip Has Sailed: A Critique of Unfounded Skepticism Around AI for Chip Design
Goldie, Anna, Mirhoseini, Azalia, Dean, Jeff
In 2020, we introduced a deep reinforcement learning method capable of generating superhuman chip layouts, which we then published in Nature and open-sourced on GitHub. AlphaChip has inspired an explosion of work on AI for chip design, and has been deployed in state-of-the-art chips across Alphabet and extended by external chipmakers. Even so, a non-peer-reviewed invited paper at ISPD 2023 questioned its performance claims, despite failing to run our method as described in Nature. For example, it did not pre-train the RL method (removing its ability to learn from prior experience), used substantially fewer compute resources (20x fewer RL experience collectors and half as many GPUs), did not train to convergence (standard practice in machine learning), and evaluated on test cases that are not representative of modern chips. Recently, Igor Markov published a meta-analysis of three papers: our peer-reviewed Nature paper, the non-peer-reviewed ISPD paper, and Markov's own unpublished paper (though he does not disclose that he co-authored it). Although AlphaChip has already achieved widespread adoption and impact, we publish this response to ensure that no one is wrongly discouraged from innovating in this impactful area.
Placement Optimization with Deep Reinforcement Learning
Goldie, Anna, Mirhoseini, Azalia
Placement Optimization is an important problem in systems and chip design, which consists of mapping the nodes of a graph onto a limited set of resources to optimize for an objective, subject to constraints. In this paper, we start by motivating reinforcement learning as a solution to the placement problem. We then give an overview of what deep reinforcement learning is. We next formulate the placement problem as a reinforcement learning problem and show how this problem can be solved with policy gradient optimization. Finally, we describe lessons we have learned from training deep reinforcement learning policies across a variety of placement optimization problems.
Azalia Mirhoseini
Azalia Mirhoseini, a research scientist at Google Brain, is using artificial intelligence itself to make better chips for artificial intelligence. Many microchips that are used for AI weren't specifically built for it. Most are repurposed from hardware used in video and gaming. As a result, these older, human-engineered designs leave much to be desired in terms of energy efficiency, cost, and functionality. Mirhoseini's system--which trained itself using trial and error, based on the AI concept of reinforcement learning--can produce chip designs in just a few hours.